package io.gitee.sfch.chunmi.common.tool;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.stream.Collectors;

/**
 * 集合工具类补全，集合可支持null.
 */
public abstract class CollectionTool {

    /**
     * 判断集合是否为空
     *
     * @param values 集合
     */
    public static boolean isEmpty(Collection<?> values) {
        return values == null || values.isEmpty();
    }

    /**
     * 判断集合是否不为空
     *
     * @param values 集合
     */
    public static boolean isNotEmpty(Collection<?> values) {
        return !isEmpty(values);
    }

    /**
     * 集合转换为Map
     *
     * @param values 集合
     * @param keyMapper KEY获取
     * @param <K> 键
     * @param <V> 值
     * @return {@link HashMap}
     */
    public static <K, V> Map<K, V> toMap(Collection<V> values, Function<V, K> keyMapper) {
        return toMap(values, keyMapper, k -> k);
    }

    /**
     * 集合转换为Map
     *
     * @param values 集合
     * @param keyMapper KEY获取
     * @param valueMapper 值获取
     * @param <K> 键
     * @param <V> 值
     * @param <E> 元素类型
     * @return {@link HashMap}
     */
    public static <K, V, E> Map<K, V> toMap(Collection<? extends E> values, Function<E, K> keyMapper,
                                            Function<E, V> valueMapper) {
        Map<K, V> map = new HashMap<>();
        if (isEmpty(values)) {
            return map;
        }
        for (E value : values) {
            K key = keyMapper.apply(value);
            map.put(key, valueMapper.apply(value));
        }
        return map;
    }


    /**
     * 将集合分组，返回分组后的Map
     *
     * @param values 集合
     * @param container 指定的Map
     * @param keyMapper 键mapper
     * @param <K> 键类型
     * @param <T> 返回的集合元素类型
     * @param <C> 参数的集合类型，需要继承于返回的集合元素类型
     * @return 将结果放入container参数内返回
     */
    public static <K, T, C extends T> Map<K, List<T>> groupBy(Collection<C> values, Map<K, List<T>> container,
                                                              Function<C, K> keyMapper) {
        if (isNotEmpty(values)) {
            for (C item : values) {
                K key = keyMapper.apply(item);
                List<T> list = container.computeIfAbsent(key, k -> new ArrayList<>());
                list.add(item);
            }
        }
        return container;
    }

    /**
     * 将集合分组，返回分组后的Map，使用HashMap存储
     *
     * @param collection 集合
     * @param keyMapper 键mapper
     * @param <K> 键类型
     * @param <T> 返回的集合元素类型
     * @param <C> 参数的集合类型，需要继承于返回的集合元素类型
     */
    public static <K, T, C extends T> Map<K, List<T>> groupBy(Collection<C> collection, Function<C, K> keyMapper) {
        return groupBy(collection, new HashMap<>(), keyMapper);
    }

    /**
     * 集合过滤，返回过滤后的数据
     *
     * @param values 集合
     * @param predicate 校验方法，返回True时保留
     * @param <T> 类型
     */
    public static <T> List<T> filter(Collection<? extends T> values, Predicate<T> predicate) {
        if (values == null) {
            return null;
        }
        return values.stream().filter(predicate).collect(Collectors.toList());
    }

    /**
     * 根据参数进行去重，若重复按顺序取第一次的值
     *
     * @param values 集合
     * @param keyMapper 唯一键获取
     * @param <E> 元素
     */
    public static <E> List<E> distinct(Collection<? extends E> values, Function<E, String> keyMapper) {
        if (isEmpty(values)) {
            return new ArrayList<>();
        }
        Set<String> keySet = new HashSet<>();
        return filter(values, k -> {
            String elementKey = keyMapper.apply(k);
            if (keySet.contains(elementKey)) {
                return false;
            }
            keySet.add(elementKey);
            return true;
        });
    }
}
